--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------
--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------
Node / ReticulumProjects / MeshChatX.git / files / tests / backend / test_nomadnet_downloader_boost.py
Displaying Raw • Download
tests/backend/test_nomadnet_downloader_boost.py v4.7.0-rc.4 (448cd46c) Text, 6.25 KB
T8b949e# SPDX-License-Identifier: 0BSD
Tff7b72import T7ee787threading
Tff7b72from T7ee787unittestT7ee787.T7ee787mock Tff7b72import Te6edf3MagicMockTb4b4b4, Te6edf3patch
Tff7b72import T7ee787pytest
Tff7b72import T7ee787RNS
Tff7b72from T7ee787meshchatxT7ee787.T7ee787srcT7ee787.T7ee787backendT7ee787.T7ee787nomadnet_downloader Tff7b72import Tb4b4b4(
Te6edf3NomadnetDownloaderTb4b4b4,
Te6edf3NomadnetFileDownloaderTb4b4b4,
Te6edf3NomadnetPageDownloaderTb4b4b4,
Te6edf3_nomadnet_links_lockTb4b4b4,
Te6edf3get_cached_active_linkTb4b4b4,
Te6edf3nomadnet_cached_linksTb4b4b4,
Tb4b4b4)
Tf0883e@pytestTff7b72.Td2a8fffixtureTb4b4b4(Te6edf3autouseTff7b72=Tff7b72TrueTb4b4b4)
Tff7b72def Td2a8ffclear_nomadnet_link_cacheTb4b4b4(Tb4b4b4)Tb4b4b4:
Tff7b72with Te6edf3_nomadnet_links_lockTb4b4b4:
Te6edf3nomadnet_cached_linksTff7b72.Td2a8ffclearTb4b4b4(Tb4b4b4)
Tff7b72yield
Tff7b72with Te6edf3_nomadnet_links_lockTb4b4b4:
Te6edf3nomadnet_cached_linksTff7b72.Td2a8ffclearTb4b4b4(Tb4b4b4)
Tf0883e@pytestTff7b72.Td2a8fffixture
Tff7b72def Td2a8ffdownloaderTb4b4b4(Tb4b4b4)Tb4b4b4:
Te6edf3on_success Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3on_failure Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3on_progress Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Tff7b72return Te6edf3NomadnetDownloaderTb4b4b4(
Ta5d6ffbTa5d6ff"Ta5d6ffdestTa5d6ff"Tb4b4b4,
Ta5d6ff"Ta5d6ff/pathTa5d6ff"Tb4b4b4,
Ta5d6ff"Ta5d6ffdataTa5d6ff"Tb4b4b4,
Te6edf3on_successTb4b4b4,
Te6edf3on_failureTb4b4b4,
Te6edf3on_progressTb4b4b4,
Tb4b4b4)
Tff7b72def Td2a8fftest_downloader_initTb4b4b4(Te6edf3downloaderTb4b4b4)Tb4b4b4:
Tff7b72assert Te6edf3downloaderTff7b72.Td2a8ffdestination_hash Tff7b72== Ta5d6ffbTa5d6ff"Ta5d6ffdestTa5d6ff"
Tff7b72assert Te6edf3downloaderTff7b72.Td2a8ffpath Tff7b72== Ta5d6ff"Ta5d6ff/pathTa5d6ff"
Tff7b72assert Te6edf3downloaderTff7b72.Td2a8ffis_cancelled Tff7b72is Tff7b72False
Tff7b72def Td2a8fftest_downloader_cancelTb4b4b4(Te6edf3downloaderTb4b4b4)Tb4b4b4:
Te6edf3downloaderTff7b72.Td2a8ffcancelTb4b4b4(Tb4b4b4)
Tff7b72assert Te6edf3downloaderTff7b72.Td2a8ffis_cancelled Tff7b72is Tff7b72True
Te6edf3downloaderTff7b72.Td2a8ff_download_failure_callbackTff7b72.Td2a8ffassert_called_withTb4b4b4(Ta5d6ff"Ta5d6ffcancelledTa5d6ff"Tb4b4b4)
Tff7b72def Td2a8fftest_cancel_removes_link_from_cacheTb4b4b4(Tb4b4b4)Tb4b4b4:
Te6edf3mock_link Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3mock_linkTff7b72.Td2a8ffstatus Tff7b72= Te6edf3RNSTff7b72.Td2a8ffLinkTff7b72.Td2a8ffACTIVE
Tff7b72with Te6edf3_nomadnet_links_lockTb4b4b4:
Te6edf3nomadnet_cached_linksTb4b4b4[Ta5d6ffbTa5d6ff"Ta5d6ffxTa5d6ff"Tb4b4b4] Tff7b72= Te6edf3mock_link
Te6edf3on_success Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3on_failure Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3on_progress Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3d Tff7b72= Te6edf3NomadnetDownloaderTb4b4b4(Ta5d6ffbTa5d6ff"Ta5d6ffxTa5d6ff"Tb4b4b4, Ta5d6ff"Ta5d6ff/pTa5d6ff"Tb4b4b4, Tff7b72NoneTb4b4b4, Te6edf3on_successTb4b4b4, Te6edf3on_failureTb4b4b4, Te6edf3on_progressTb4b4b4)
Te6edf3dTff7b72.Td2a8fflink Tff7b72= Te6edf3mock_link
Te6edf3dTff7b72.Td2a8ffcancelTb4b4b4(Tb4b4b4)
Tff7b72assert Te6edf3get_cached_active_linkTb4b4b4(Ta5d6ffbTa5d6ff"Ta5d6ffxTa5d6ff"Tb4b4b4) Tff7b72is Tff7b72None
Te6edf3mock_linkTff7b72.Td2a8ffteardownTff7b72.Td2a8ffassert_called_onceTb4b4b4(Tb4b4b4)
Tff7b72def Td2a8fftest_get_cached_active_link_evicts_staleTb4b4b4(Tb4b4b4)Tb4b4b4:
Te6edf3dead Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3deadTff7b72.Td2a8ffstatus Tff7b72= Tff7b72None
Tff7b72with Te6edf3_nomadnet_links_lockTb4b4b4:
Te6edf3nomadnet_cached_linksTb4b4b4[Ta5d6ffbTa5d6ff"Ta5d6ffzTa5d6ff"Tb4b4b4] Tff7b72= Te6edf3dead
Tff7b72assert Te6edf3get_cached_active_linkTb4b4b4(Ta5d6ffbTa5d6ff"Ta5d6ffzTa5d6ff"Tb4b4b4) Tff7b72is Tff7b72None
Tff7b72with Te6edf3_nomadnet_links_lockTb4b4b4:
Tff7b72assert Ta5d6ffbTa5d6ff"Ta5d6ffzTa5d6ff" Tff7b72not Tff7b72in Te6edf3nomadnet_cached_links
Tf0883e@pytestTff7b72.Td2a8ffmarkTff7b72.Td2a8ffasyncio
Tff7b72async Tff7b72def Td2a8fftest_download_no_pathTb4b4b4(Te6edf3downloaderTb4b4b4)Tb4b4b4:
Tff7b72with Tb4b4b4(
Te6edf3patchTff7b72.Td2a8ffobjectTb4b4b4(Te6edf3RNSTff7b72.Td2a8ffTransportTb4b4b4, Ta5d6ff"Ta5d6ffhas_pathTa5d6ff"Tb4b4b4, Te6edf3return_valueTff7b72=Tff7b72FalseTb4b4b4)Tb4b4b4,
Te6edf3patchTff7b72.Td2a8ffobjectTb4b4b4(Te6edf3RNSTff7b72.Td2a8ffTransportTb4b4b4, Ta5d6ff"Ta5d6ffrequest_pathTa5d6ff"Tb4b4b4)Tb4b4b4,
Tb4b4b4)Tb4b4b4:
Tff7b72await Te6edf3downloaderTff7b72.Td2a8ffdownloadTb4b4b4(Te6edf3path_lookup_timeoutTff7b72=T79c0ff0.1Tb4b4b4)
Te6edf3downloaderTff7b72.Td2a8ff_download_failure_callbackTff7b72.Td2a8ffassert_called_withTb4b4b4(
Ta5d6ff"Ta5d6ffCould not find path to destination.Ta5d6ff"Tb4b4b4,
Tb4b4b4)
Tf0883e@pytestTff7b72.Td2a8ffmarkTff7b72.Td2a8ffasyncio
Tff7b72async Tff7b72def Td2a8fftest_download_cached_linkTb4b4b4(Te6edf3downloaderTb4b4b4)Tb4b4b4:
Te6edf3mock_link Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3mock_linkTff7b72.Td2a8ffstatus Tff7b72= Te6edf3RNSTff7b72.Td2a8ffLinkTff7b72.Td2a8ffACTIVE
Tff7b72with Te6edf3_nomadnet_links_lockTb4b4b4:
Te6edf3nomadnet_cached_linksTb4b4b4[Ta5d6ffbTa5d6ff"Ta5d6ffdestTa5d6ff"Tb4b4b4] Tff7b72= Te6edf3mock_link
Tff7b72with Te6edf3patchTff7b72.Td2a8ffobjectTb4b4b4(Te6edf3downloaderTb4b4b4, Ta5d6ff"Ta5d6fflink_establishedTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3mock_establishedTb4b4b4:
Tff7b72await Te6edf3downloaderTff7b72.Td2a8ffdownloadTb4b4b4(Tb4b4b4)
Te6edf3mock_establishedTff7b72.Td2a8ffassert_called_withTb4b4b4(Te6edf3mock_linkTb4b4b4)
Tf0883e@pytestTff7b72.Td2a8ffmarkTff7b72.Td2a8ffasyncio
Tff7b72async Tff7b72def Td2a8fftest_download_uses_path_wait_cache_hitTb4b4b4(Te6edf3downloaderTb4b4b4)Tb4b4b4:
T8b949e"""Another task may populate the cache while we wait for a path."""
Te6edf3mock_link Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3mock_linkTff7b72.Td2a8ffstatus Tff7b72= Te6edf3RNSTff7b72.Td2a8ffLinkTff7b72.Td2a8ffACTIVE
Te6edf3call_count Tff7b72= Tb4b4b4{Ta5d6ff"Ta5d6ffnTa5d6ff"Tb4b4b4: T79c0ff0Tb4b4b4}
Tff7b72def Td2a8ffhas_path_side_effectTb4b4b4(Te6edf3destTb4b4b4)Tb4b4b4:
Te6edf3call_countTb4b4b4[Ta5d6ff"Ta5d6ffnTa5d6ff"Tb4b4b4] Tff7b72+Tff7b72= T79c0ff1
Tff7b72if Te6edf3call_countTb4b4b4[Ta5d6ff"Ta5d6ffnTa5d6ff"Tb4b4b4] Tff7b72== T79c0ff1Tb4b4b4:
Tff7b72return Tff7b72False
Tff7b72if Te6edf3call_countTb4b4b4[Ta5d6ff"Ta5d6ffnTa5d6ff"Tb4b4b4] Tff7b72== T79c0ff2Tb4b4b4:
Tff7b72with Te6edf3_nomadnet_links_lockTb4b4b4:
Te6edf3nomadnet_cached_linksTb4b4b4[Te6edf3destTb4b4b4] Tff7b72= Te6edf3mock_link
Tff7b72return Tff7b72True
Tff7b72with Tb4b4b4(
Te6edf3patchTff7b72.Td2a8ffobjectTb4b4b4(Te6edf3RNSTff7b72.Td2a8ffTransportTb4b4b4, Ta5d6ff"Ta5d6ffhas_pathTa5d6ff"Tb4b4b4, Te6edf3side_effectTff7b72=Te6edf3has_path_side_effectTb4b4b4)Tb4b4b4,
Te6edf3patchTff7b72.Td2a8ffobjectTb4b4b4(Te6edf3RNSTff7b72.Td2a8ffTransportTb4b4b4, Ta5d6ff"Ta5d6ffrequest_pathTa5d6ff"Tb4b4b4)Tb4b4b4,
Tb4b4b4)Tb4b4b4:
Tff7b72with Te6edf3patchTff7b72.Td2a8ffobjectTb4b4b4(Te6edf3downloaderTb4b4b4, Ta5d6ff"Ta5d6fflink_establishedTa5d6ff"Tb4b4b4) Tff7b72as Te6edf3mock_establishedTb4b4b4:
Tff7b72await Te6edf3downloaderTff7b72.Td2a8ffdownloadTb4b4b4(
Te6edf3path_lookup_timeoutTff7b72=T79c0ff5Tb4b4b4,
Te6edf3link_establishment_timeoutTff7b72=T79c0ff5Tb4b4b4,
Tb4b4b4)
Te6edf3mock_establishedTff7b72.Td2a8ffassert_called_once_withTb4b4b4(Te6edf3mock_linkTb4b4b4)
Tff7b72def Td2a8fftest_page_downloader_invalid_utf8_replacedTb4b4b4(Tb4b4b4)Tb4b4b4:
Te6edf3on_ok Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3on_fail Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3on_progress Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3pd Tff7b72= Te6edf3NomadnetPageDownloaderTb4b4b4(
Ta5d6ffbTa5d6ff"Ta5d6ffabTa5d6ff" Tff7b72* T79c0ff8Tb4b4b4,
Ta5d6ff"Ta5d6ff/page.muTa5d6ff"Tb4b4b4,
Tff7b72NoneTb4b4b4,
Te6edf3on_okTb4b4b4,
Te6edf3on_failTb4b4b4,
Te6edf3on_progressTb4b4b4,
Tb4b4b4)
Te6edf3rr Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3rrTff7b72.Td2a8ffresponse Tff7b72= Ta5d6ffbTa5d6ff"Ta5d6ffhelloTffea00\xffTffea00\xfeTa5d6ffinvalidTa5d6ff"
Te6edf3pdTff7b72.Td2a8ffon_download_successTb4b4b4(Te6edf3rrTb4b4b4)
Te6edf3on_okTff7b72.Td2a8ffassert_called_onceTb4b4b4(Tb4b4b4)
Tff7b72assert Ta5d6ff"Tffea00\ufffdTa5d6ff" Tff7b72in Te6edf3on_okTff7b72.Td2a8ffcall_argsTb4b4b4[T79c0ff0Tb4b4b4]Tb4b4b4[T79c0ff0Tb4b4b4]
Te6edf3on_failTff7b72.Td2a8ffassert_not_calledTb4b4b4(Tb4b4b4)
Tff7b72def Td2a8fftest_page_downloader_empty_responseTb4b4b4(Tb4b4b4)Tb4b4b4:
Te6edf3on_ok Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3on_fail Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3pd Tff7b72= Te6edf3NomadnetPageDownloaderTb4b4b4(
Ta5d6ffbTa5d6ff"Ta5d6ffabTa5d6ff" Tff7b72* T79c0ff8Tb4b4b4,
Ta5d6ff"Ta5d6ff/page.muTa5d6ff"Tb4b4b4,
Tff7b72NoneTb4b4b4,
Te6edf3on_okTb4b4b4,
Te6edf3on_failTb4b4b4,
Te6edf3MagicMockTb4b4b4(Tb4b4b4)Tb4b4b4,
Tb4b4b4)
Te6edf3rr Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3rrTff7b72.Td2a8ffresponse Tff7b72= Tff7b72None
Te6edf3pdTff7b72.Td2a8ffon_download_successTb4b4b4(Te6edf3rrTb4b4b4)
Te6edf3on_failTff7b72.Td2a8ffassert_called_once_withTb4b4b4(Ta5d6ff"Ta5d6ffempty_responseTa5d6ff"Tb4b4b4)
Te6edf3on_okTff7b72.Td2a8ffassert_not_calledTb4b4b4(Tb4b4b4)
Tff7b72def Td2a8fftest_file_downloader_list_response_short_list_no_crashTb4b4b4(Tb4b4b4)Tb4b4b4:
Te6edf3on_ok Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3on_fail Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3fd Tff7b72= Te6edf3NomadnetFileDownloaderTb4b4b4(
Ta5d6ffbTa5d6ff"Ta5d6ffabTa5d6ff" Tff7b72* T79c0ff8Tb4b4b4,
Ta5d6ff"Ta5d6ff/f.binTa5d6ff"Tb4b4b4,
Te6edf3on_okTb4b4b4,
Te6edf3on_failTb4b4b4,
Te6edf3MagicMockTb4b4b4(Tb4b4b4)Tb4b4b4,
Tb4b4b4)
Te6edf3rr Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3rrTff7b72.Td2a8ffresponse Tff7b72= Tb4b4b4[Ta5d6ffbTa5d6ff"Ta5d6ffonlyTa5d6ff"Tb4b4b4]
Te6edf3fdTff7b72.Td2a8ffon_download_successTb4b4b4(Te6edf3rrTb4b4b4)
Te6edf3on_failTff7b72.Td2a8ffassert_called_once_withTb4b4b4(Ta5d6ff"Ta5d6ffunsupported_responseTa5d6ff"Tb4b4b4)
Tff7b72def Td2a8fftest_file_downloader_stores_data_parameterTb4b4b4(Tb4b4b4)Tb4b4b4:
Te6edf3on_ok Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3on_fail Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3on_progress Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3fd Tff7b72= Te6edf3NomadnetFileDownloaderTb4b4b4(
Ta5d6ffbTa5d6ff"Ta5d6ffabTa5d6ff" Tff7b72* T79c0ff8Tb4b4b4,
Ta5d6ff"Ta5d6ff/file/data.binTa5d6ff"Tb4b4b4,
Te6edf3on_okTb4b4b4,
Te6edf3on_failTb4b4b4,
Te6edf3on_progressTb4b4b4,
Te6edf3dataTff7b72=Ta5d6ff"Ta5d6ffquery=value&other=123Ta5d6ff"Tb4b4b4,
Tb4b4b4)
Tff7b72assert Te6edf3fdTff7b72.Td2a8ffdata Tff7b72== Ta5d6ff"Ta5d6ffquery=value&other=123Ta5d6ff"
Tff7b72def Td2a8fftest_file_downloader_passes_data_to_parentTb4b4b4(Tb4b4b4)Tb4b4b4:
Te6edf3on_ok Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3on_fail Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3on_progress Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3fd Tff7b72= Te6edf3NomadnetFileDownloaderTb4b4b4(
Ta5d6ffbTa5d6ff"Ta5d6ffabTa5d6ff" Tff7b72* T79c0ff8Tb4b4b4,
Ta5d6ff"Ta5d6ff/file/data.binTa5d6ff"Tb4b4b4,
Te6edf3on_okTb4b4b4,
Te6edf3on_failTb4b4b4,
Te6edf3on_progressTb4b4b4,
Te6edf3dataTff7b72=Ta5d6ff"Ta5d6fffoo=barTa5d6ff"Tb4b4b4,
Tb4b4b4)
T8b949e# NomadnetDownloader stores data as the 3rd positional arg
Tff7b72assert Te6edf3fdTff7b72.Td2a8ffdata Tff7b72== Ta5d6ff"Ta5d6fffoo=barTa5d6ff"
Tff7b72def Td2a8fftest_cache_lock_serializes_mutationsTb4b4b4(Tb4b4b4)Tb4b4b4:
Te6edf3mock_link Tff7b72= Te6edf3MagicMockTb4b4b4(Tb4b4b4)
Te6edf3mock_linkTff7b72.Td2a8ffstatus Tff7b72= Te6edf3RNSTff7b72.Td2a8ffLinkTff7b72.Td2a8ffACTIVE
Te6edf3errors Tff7b72= Tb4b4b4[Tb4b4b4]
Tff7b72def Td2a8ffworkerTb4b4b4(Tb4b4b4)Tb4b4b4:
Tff7b72tryTb4b4b4:
Tff7b72with Te6edf3_nomadnet_links_lockTb4b4b4:
Te6edf3nomadnet_cached_linksTb4b4b4[Ta5d6ffbTa5d6ff"Ta5d6fftTa5d6ff"Tb4b4b4] Tff7b72= Te6edf3mock_link
Tff7b72assert Te6edf3nomadnet_cached_linksTb4b4b4[Ta5d6ffbTa5d6ff"Ta5d6fftTa5d6ff"Tb4b4b4] Tff7b72is Te6edf3mock_link
Tff7b72except Tf85149Exception Tff7b72as Te6edf3eTb4b4b4:
Te6edf3errorsTff7b72.Td2a8ffappendTb4b4b4(Te6edf3eTb4b4b4)
Te6edf3threads Tff7b72= Tb4b4b4[Te6edf3threadingTff7b72.Td2a8ffThreadTb4b4b4(Te6edf3targetTff7b72=Te6edf3workerTb4b4b4) Tff7b72for Te6edf3_ Tff7b72in Tffa657rangeTb4b4b4(T79c0ff8Tb4b4b4)Tb4b4b4]
Tff7b72for Te6edf3t Tff7b72in Te6edf3threadsTb4b4b4:
Te6edf3tTff7b72.Td2a8ffstartTb4b4b4(Tb4b4b4)
Tff7b72for Te6edf3t Tff7b72in Te6edf3threadsTb4b4b4:
Te6edf3tTff7b72.Td2a8ffjoinTb4b4b4(Tb4b4b4)
Tff7b72assert Tff7b72not Te6edf3errors
──────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────